package simple.linklist;

import struct.ListNode;

/**
 * 203. 移除链表元素
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 * 示例 1：
 *   输入：head = [1,2,6,3,4,5,6], val = 6
 *   输出：[1,2,3,4,5]
 * 示例 2：
 *   输入：head = [], val = 1
 *   输出：[]
 * 示例 3：
 *   输入：head = [7,7,7,7], val = 7
 *   输出：[]
 * 提示：
 *   列表中的节点数目在范围 [0, 104] 内
 *   1 <= Node.val <= 50
 *   0 <= val <= 50
 * @author 刘学松
 * @date 2022-03-07 20:44
 */
public class 移除链表元素 {
    /**
     * 迭代
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements(ListNode head, int val) {
        ListNode dummyNode = new ListNode(val-1);
        dummyNode.next = head;
        ListNode tempNode = dummyNode;
        while (tempNode.next != null) {
            if (tempNode.next.val == val) {
                tempNode.next = tempNode.next.next;
            } else {
                tempNode = tempNode.next;
            }
        }
        return dummyNode.next;
    }

    /**
     * 递归
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements1(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        head.next = removeElements(head.next, val);
        return head.val == val ? head.next : head;
    }
}
